import PyInstaller.__main__
import os

def build_exe():
    # 获取当前目录
    current_dir = os.path.dirname(os.path.abspath(__file__))
    
    # 定义图标文件路径
    icon_path = os.path.join(current_dir, 'icon.ico')
    
    # 定义打包参数
    params = [
        'main.py',  # 主程序文件
        '--name=文件统计工具',  # 生成的exe名称
        '--windowed',  # 使用GUI模式
        '--clean',  # 清理临时文件
        '--noconfirm',  # 不确认覆盖
        '--onefile',  # 打包成单个文件
        '--add-data=README.md;.',  # 添加额外文件
        f'--icon={icon_path}',  # 添加图标
        '--hidden-import=pandas',
        '--hidden-import=openpyxl',
        '--hidden-import=PyQt6',
        '--hidden-import=PyQt6.QtCore',
        '--hidden-import=PyQt6.QtGui',
        '--hidden-import=PyQt6.QtWidgets',
    ]
    
    # 执行打包命令
    PyInstaller.__main__.run(params)

if __name__ == '__main__':
    build_exe() 